import pandas as pd

# 读取原始数据（请替换为您的文件路径）
input_path = "D://laiyifen//门店信息.xlsx"
df = pd.read_excel(input_path)

# 更新后的列名映射字典
column_mapping = {
    "原始门店状态列名": "门店状态",
    "SAP门店编码": "费用使用方",          # 修改点1
    "原始门店名称列名": "门店名称",
    "门店类型": "门店性质（直营/加盟）",  # 修改点2
    "省份代码": "地区",              # 修改点3
    "原始销售组织列名": "销售组织",
    "原始管理公司列名": "所属管理公司名称",
    "原始加盟商代码列名": "加盟商代码",
    "原始加盟商名称列名": "加盟商名称",
    "原始核算组织列名": "核算组织代码",
    "成本中心编码": "成本中心代码"        # 修改点4
}

# 重命名列
df.rename(columns=column_mapping, inplace=True)

# 定义有效列（更新列名）
valid_columns = [
    "门店状态",
    "费用使用方",
    "门店名称",
    "门店性质（直营/加盟）",
    "地区",
    "销售组织",
    "所属管理公司名称",
    "加盟商代码",
    "加盟商名称",
    "核算组织代码",
    "成本中心代码"
]

# 数据筛选（更新列名）
valid_provinces = [
    "安徽省", "北京", "甘肃省", "贵州省", "海南省",
    "黑龙江省", "吉林省", "江苏省", "辽宁省", "内蒙古自治区",
    "宁夏回族自治区", "青海省", "天津", "西藏自治区",
    "新疆维吾尔自治区", "云南省", "重庆"
]

filtered_df = df[
    (df["地区"].isin(valid_provinces))  # 使用新列名
][valid_columns]

# 调整列顺序（更新列名）
new_column_order = [
    "门店状态",           # A列
    "费用使用方",         # C列（原SAP门店编码）
    "门店名称",           # D列
    "门店性质（直营/加盟）", # E列
    "地区",               # F列（原省份代码）
    "销售组织",           # L列
    "所属管理公司名称",     # N列
    "加盟商代码",         # P列
    "加盟商名称",         # R列
    "核算组织代码",       # S列
    "成本中心代码"        # T列（原成本中心编码）
]

# 最终处理结果
result_df = filtered_df[new_column_order]

# 添加新列（保持不变）
result_df["所属子公司-直营"] = "C" + result_df["销售组织"] + "-" + result_df["所属管理公司名称"]
result_df["所属子公司-加盟"] = result_df["加盟商代码"] + "-" + result_df["加盟商名称"]

# 更新最终列顺序
final_columns = new_column_order + ["所属子公司-直营", "所属子公司-加盟"]
result_df = result_df[final_columns]

# 保存结果（请修改输出路径）
output_path = "D://laiyifen//2.xlsx"
result_df.to_excel(output_path, index=False)

print("处理完成！结果已保存至:", output_path)